/*
Date：20210414 12：00pm
key:1.相当于循环合并两个链表，不过最后一个测试点超时。
 */
/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
        ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
        if(l1==nullptr||l2==nullptr)
        {
            if(l1==nullptr&&l2==nullptr)
            {
                return l1;
            }
            if(l1==nullptr)
            {
                return l2;
            }
            else
            {
                return l1;
            }
        }
        int count=0;
        ListNode* head=new ListNode;
        ListNode* p=head;
        while(l1!=nullptr&&l2!=nullptr)
        {
            if(l1->val>l2->val)
            {
                p->val=l2->val;
                l2=l2->next;
            }
            else
            {
                p->val=l1->val;
                l1=l1->next;
            }
            count++;
            p->next=new ListNode;;
            p=p->next;
        }
        while(l1!=nullptr)
        {

            p->val=l1->val;
            l1=l1->next;
            count++;
            p->next=new ListNode;;
            p=p->next;
        }
        while(l2!=nullptr)
        {

            p->val=l2->val;
            l2=l2->next;
            count++;
            p->next=new ListNode;;
            p=p->next;
        }
        p=head;
        
        while(--count)
        {
            p=p->next;
        }
        p->next=nullptr;
        return head;
    }
    ListNode* mergeKLists(vector<ListNode*>& lists)
    {
        for(int i=0;i<lists.size();i++)
        {
            if(lists[i]==nullptr)
            {
                lists.erase(lists.begin()+i);
            }
        }
        if(lists.size()==0)
        {
            return nullptr;
        }

        ListNode*p;
        for(int i=0;i<lists.size();i++)
        {
            if(i==0)
            {
                p=lists[0];
            }
            else
            {
                p=mergeTwoLists(p,lists[i]);
            }
        }
        return p;
    }
};